from alg import restore_ip_addresses
from gen import generate_ip_test_string

def test():
    # 随机生成测试字符串
    test_string = generate_ip_test_string()
    print(f"测试字符串: {test_string}")
    result = restore_ip_addresses(test_string)
    print(f"可能的IP地址数量: {len(result)}")
    print(f"所有可能的IP地址: {result}")

    # 特定测试用例
    test_cases = [
        "25525511135",   
        "0000",        
        "1111",        
        "010010",       
        "101023"         
    ]
    
    print("\n特定测试用例:")
    for s in test_cases:
        result = restore_ip_addresses(s)
        print(f"字符串 = {s}, 可能的IP地址: {result}")

if __name__ == "__main__":
    test()